/*****************************************************************************
 *
 * This MobilityDB code is provided under The PostgreSQL License.
 * Copyright (c) 2016-2024, Université libre de Bruxelles and MobilityDB
 * contributors
 *
 * MobilityDB includes portions of PostGIS version 3 source code released
 * under the GNU General Public License (GPLv2 or later).
 * Copyright (c) 2001-2024, PostGIS contributors
 *
 * Permission to use, copy, modify, and distribute this software and its
 * documentation for any purpose, without fee, and without a written
 * agreement is hereby granted, provided that the above copyright notice and
 * this paragraph and the following two paragraphs appear in all copies.
 *
 * IN NO EVENT SHALL UNIVERSITE LIBRE DE BRUXELLES BE LIABLE TO ANY PARTY FOR
 * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING
 * LOST PROFITS, ARISING OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION,
 * EVEN IF UNIVERSITE LIBRE DE BRUXELLES HAS BEEN ADVISED OF THE POSSIBILITY
 * OF SUCH DAMAGE.
 *
 * UNIVERSITE LIBRE DE BRUXELLES SPECIFICALLY DISCLAIMS ANY WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE. THE SOFTWARE PROVIDED HEREUNDER IS ON
 * AN "AS IS" BASIS, AND UNIVERSITE LIBRE DE BRUXELLES HAS NO OBLIGATIONS TO
 * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
 *
 *****************************************************************************/

/**
 * @brief A simple program that reads from a CSV file synthetic trip data in
 * Brussels generated by the MobilityDB-BerlinMOD generator
 * https://github.com/MobilityDB/MobilityDB-BerlinMOD
 * splits the trips and the speed of these trips by tiles, and computes
 * aggregate values of the count, duration, and distance of the tiles.
 *
 * Please read the assumptions made about the input file in the file
 * `05_berlinmod_disassemble.c` in the same directory. Furthermore, the program
 * that we already have computed the extent of the trips and the speeds.
 * Otherwise, a first scan of the CSV file is needed for computing both extents
 * and then proceed for the second scan as below.
 *
 * The program can be build as follows
 * @code
 * gcc -Wall -g -I/usr/local/include -o 07_berlinmod_tile 07_berlinmod_tile.c -L/usr/local/lib -lmeos
 * @endcode
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <meos.h>

/* Maximum length in characters of a trip in the input data */
#define MAX_LENGTH_TRIP 170001
/* Maximum length in characters of a header record in the input CSV file */
#define MAX_LENGTH_HEADER 1024
/* Maximum length in characters of a date in the input data */
#define MAX_LENGTH_DATE 12
/* Maximum number of bins in each dimension */
#define MAX_NO_BINS 10

typedef struct
{
  int count;
  Interval duration;
  double distance;
} trip_record;

typedef struct
{
  int count;
  Interval duration;
} speed_record;

/* Main program */
int main(void)
{
  /* Variables to read the input CSV file */
  char trip_buffer[MAX_LENGTH_TRIP];
  char header_buffer[MAX_LENGTH_HEADER];
  char date_buffer[MAX_LENGTH_DATE];
  /* Number of records */
  int no_records = 0;
  int no_nulls = 0;
  /* Iterator variables */
  int i, k;

  /* Get start time */
  clock_t t;
  t = clock();

  /* Initialize MEOS */
  meos_initialize();

  /* Compute the spatial tiles for trips */
  int no_speed_tiles, no_trip_tiles;
  STBox *trip_extent =
    stbox_in("SRID=3857;STBOX X((473212.810151,6578740.528027),(499152.544688,6607165.513683))");
  GSERIALIZED *sorigin = geom_in("SRID=3857;Point(0 0)", -1);
  STBox *trip_tiles = stbox_space_tiles(trip_extent, 5e3, 5e3, 0, sorigin,
    true, &no_trip_tiles);
  /* Compute the (value and time) tiles for speed of trips */
  TBox *speed_extent = tbox_in("TBox XT([0, 35),[2020-06-01, 2020-06-05))");
  Interval *duration = interval_in("1 day", -1);
  TimestampTz torigin = timestamptz_in("2020-06-01", -1);
  TBox *speed_tiles = tfloatbox_value_time_tiles(speed_extent, 10.0, duration,
    0.0, torigin, &no_speed_tiles);
  /* Variables for tiling the trips and their speeds */
  trip_record *trip_splits = malloc(sizeof(trip_record) * no_trip_tiles);
  memset(trip_splits, 0, sizeof(trip_record) * no_trip_tiles);
  speed_record *speed_splits = malloc(sizeof(speed_record) * no_speed_tiles);
  memset(speed_splits, 0, sizeof(speed_record) * no_speed_tiles);

  free(trip_extent);
  free(sorigin);
  free(speed_extent);
  free(duration);

  /* Substitute the full file path in the first argument of fopen */
  FILE *file = fopen("data/berlinmod_trips.csv", "r");

  if (! file)
  {
    printf("Error opening input file\n");
    free(trip_tiles);
    free(trip_splits);
    free(speed_tiles);
    free(speed_splits);
    return 1;
  }

  /* Read the first line of the file with the headers */
  fscanf(file, "%1023s\n", header_buffer);
  printf("Processing records (one marker per trip)\n");

  /* Continue reading the file */
  do
  {
    int tripId, vehId, seq;
    int read = fscanf(file, "%d,%d,%10[^,],%d,%170000[^\n]\n",
      &tripId, &vehId, date_buffer, &seq, trip_buffer);
    /* Transform the string representing the trip into a temporal value */
    Temporal *trip = temporal_from_hexwkb(trip_buffer);
    Temporal *speed = tpoint_speed(trip);
    Temporal *split;
    Interval *dur1, *dur2;

    if (read == 5)
    {
      no_records++;
      printf("*");
      fflush(stdout);
    }

    if (read != 5 && !feof(file))
    {
      printf("Record with missing values ignored\n");
      no_nulls++;
    }

    if (ferror(file))
    {
      printf("Error reading input file\n");
      fclose(file);
      free(trip_tiles);
      free(trip_splits);
      free(speed_tiles);
      free(speed_splits);
      return 1;
    }

    /* Split the trip by the tiles and accumulate aggregate values */
    k = 0;
    for (i = 0; i < no_trip_tiles; i++)
    {
      split = tpoint_at_stbox(trip, &trip_tiles[k], false);
      if (split != NULL)
      {
        trip_splits[k].count++;
        trip_splits[k].distance += tpoint_length(split) / 1e3;
        dur1 = temporal_duration(split, false);
        dur2 = add_interval_interval(dur1, &trip_splits[k].duration);
        memcpy(&trip_splits[k].duration, dur2, sizeof(Interval));
        free(split); free(dur1); free(dur2);
      }
      k++;
    }

    /* Split the temporal speed by the tiles and accumulate aggregate values */
    k = 0;
    for (i = 0; i < no_speed_tiles; i++)
    {
      split = tnumber_at_tbox(speed, &speed_tiles[k]);
      if (split != NULL)
      {
        speed_splits[k].count++;
        dur1 = temporal_duration(split, false);
        dur2 = add_interval_interval(dur1, &speed_splits[k].duration);
        memcpy(&speed_splits[k].duration, dur2, sizeof(Interval));
        free(split); free(dur1); free(dur2);
      }
      k++;
    }

    free(trip);
    free(speed);

  } while (!feof(file));

  printf("\n%d records read.\n%d incomplete records ignored.\n",
    no_records, no_nulls);

  /* Print results */
  printf("-------------\n");
  printf(" Trip tiles\n");
  printf("-------------\n");
  k = 0;
  Interval intervalzero;
  memset(&intervalzero, 0, sizeof(Interval));
  for (i = 0; i < no_trip_tiles; i++)
  {
    if (interval_cmp(&trip_splits[k].duration, &intervalzero) != 0)
    {
      char *stbox_str = stbox_out(&trip_tiles[k], 0);
      char *interval_str = interval_out(&trip_splits[k].duration);
      printf("Tile: %d, Box: %s, Count: %d, Duration: %s, Distance: %f\n",
        i, stbox_str, trip_splits[k].count, interval_str,
        trip_splits[k].distance);
      free(stbox_str);
      free(interval_str);
    }
    k++;
  }
  printf("\n");

  /* Print results */
  printf("-------------\n");
  printf(" Speed tiles\n");
  printf("-------------\n");
  k = 0;
  for (i = 0; i < no_speed_tiles; i++)
  {
    char *span_str = floatspan_out(&speed_tiles[k].span, 0);
    char *tstzspan_str = tstzspan_out(&speed_tiles[k].period);
    char *interval_str = interval_out(&speed_splits[k].duration);
    printf("Tile: %d, Span: %s, Period: %s, Count: %d, Duration: %s\n",
      i, span_str, tstzspan_str, speed_splits[k].count, interval_str);
    free(span_str); free(tstzspan_str); free(interval_str);
    k++;
  }

  /* Free memory */
  free(trip_tiles);
  free(trip_splits);
  free(speed_tiles);
  free(speed_splits);

  /* Close the file */
  fclose(file);

  /* Finalize MEOS */
  meos_finalize();

  /* Calculate the elapsed time */
  t = clock() - t;
  double time_taken = ((double) t) / CLOCKS_PER_SEC;
  printf("The program took %f seconds to execute\n", time_taken);

  return 0;
}
